../../../../../examples/tutorials/mnist_pytorch/data.py